# 连接数据库
import sqlite3


def get_db(query, args=(), one=False):
    db = sqlite3.connect('mydb.db')
    db.row_factory = sqlite3.Row
    cur = db.execute(query, args)
    db.commit()  # 提交数据
    rv = cur.fetchall()  # 返回多条数据，getchone：返回一条数据
    db.close()  # 关闭数据库
    return (rv[0] if rv else None) if one else rv


def query_db(query, args=()):
    db = sqlite3.connect('mydb.db')  # 创建数据库
    db.text_factory = str  # Python和SQLite类型转换不统一，将数据库连接的text_factory属性设置为str
    db.executemany(query, args)
    db.commit()
    db.close()
